SAE vs Gated SAE on toy model